from enum import Enum
from pydantic import BaseModel, Field


class TMITYPE(Enum):
    GDP = "G"
    AFP = "A"
    MIT = "M"
    GS = "S"


class NTFM_FLIGHT(BaseModel):
    """
    流控受限标识
        0：不受限
        1：受限：航班受流控措施影响，且未被豁免或
        人工指定 CTO
    """
    RESFLAG: bool = Field(..., description=""""
    0：不受限
1：受限：航班受流控措施影响，且未被豁免或
人工指定 CTOT""")
    TMITYPE: TMITYPE = Field(..., description="""
    描述该航班受哪些措施类型影响，多措施以
分号分隔。
    """)


if __name__ == '__main__':
    dd = {
        "RESFLAG": True,
        "TMITYPE": "G"
    }
    tt = NTFM_FLIGHT(**dd)
    print(tt.RESFLAG)
    print(tt.TMITYPE)
